{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "from infer_vid import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "100%|█████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:07<00:00, 707.52it/s]\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "using cosine layer\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "100%|██████████████████████████████████████████████████████████████████████████████████| 50/50 [00:53<00:00,  1.07s/it]\n",
      "100%|██████████████████████████████████████████████████████████████████████████████████| 24/24 [00:08<00:00,  2.67it/s]\n"
     ]
    }
   ],
   "source": [
    "inst_ds = get_pred_vid('../../input/validation_dataset_part1/video')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'file_name': '../../input/validation_dataset_part1/video/000919.mp4',\n",
       " 'height': 960,\n",
       " 'width': 540,\n",
       " 'item_id': '000919',\n",
       " 'type': 'video',\n",
       " 'bbox': [99, 359, 318, 814],\n",
       " 'score': 0.9992340803146362,\n",
       " 'frame': 280,\n",
       " 'bbox_aspect': 0.48131868131868133,\n",
       " 'aspect_group': 3,\n",
       " 'feat': [0.10861119627952576,\n",
       "  -0.02023276872932911,\n",
       "  0.000677834206726402,\n",
       "  0.04453166946768761,\n",
       "  -0.08077826350927353,\n",
       "  0.013814250938594341,\n",
       "  -0.008523191325366497,\n",
       "  -0.0611051470041275,\n",
       "  -0.0397714264690876,\n",
       "  0.03194702789187431,\n",
       "  -0.0006806030287407339,\n",
       "  -0.019916066899895668,\n",
       "  0.07606640458106995,\n",
       "  0.009971742518246174,\n",
       "  -0.013884657062590122,\n",
       "  -0.021979905664920807,\n",
       "  0.08041248470544815,\n",
       "  -0.0171558428555727,\n",
       "  0.01628359965980053,\n",
       "  0.027200501412153244,\n",
       "  0.010065089911222458,\n",
       "  0.01591976173222065,\n",
       "  0.0682348683476448,\n",
       "  -0.04079398885369301,\n",
       "  0.0012808083556592464,\n",
       "  0.0553743839263916,\n",
       "  -0.01839214749634266,\n",
       "  0.08164402842521667,\n",
       "  -0.05385705828666687,\n",
       "  -0.019228944554924965,\n",
       "  0.02114301547408104,\n",
       "  -0.024600272998213768,\n",
       "  -0.054915424436330795,\n",
       "  -0.1035628691315651,\n",
       "  -0.007335315458476543,\n",
       "  0.02674494870007038,\n",
       "  -0.011918866075575352,\n",
       "  0.0051943291909992695,\n",
       "  -0.037380021065473557,\n",
       "  -0.044803548604249954,\n",
       "  -0.02558865025639534,\n",
       "  -0.057260558009147644,\n",
       "  0.0363578125834465,\n",
       "  -0.012578342109918594,\n",
       "  -0.05943750590085983,\n",
       "  -0.09108768403530121,\n",
       "  -0.014429399743676186,\n",
       "  -0.016254955902695656,\n",
       "  -0.04457131400704384,\n",
       "  -0.04249356687068939,\n",
       "  -0.02692616730928421,\n",
       "  0.09313350915908813,\n",
       "  0.01826678216457367,\n",
       "  -0.015197738073766232,\n",
       "  -0.07252022624015808,\n",
       "  -0.07828399538993835,\n",
       "  0.004812547471374273,\n",
       "  0.00643581710755825,\n",
       "  -0.05307281017303467,\n",
       "  -0.0702512264251709,\n",
       "  0.04165658354759216,\n",
       "  0.03468478098511696,\n",
       "  -0.028483951464295387,\n",
       "  -0.016901392489671707,\n",
       "  -0.002983693266287446,\n",
       "  -0.023870795965194702,\n",
       "  0.03681115433573723,\n",
       "  0.017926208674907684,\n",
       "  -0.0354781448841095,\n",
       "  -0.008896398358047009,\n",
       "  0.008050533011555672,\n",
       "  0.10579269379377365,\n",
       "  -0.015211955644190311,\n",
       "  -0.003988060634583235,\n",
       "  -0.023958537727594376,\n",
       "  0.0045199329033494,\n",
       "  -0.009660933166742325,\n",
       "  0.07028080523014069,\n",
       "  -0.008721746504306793,\n",
       "  0.003815128467977047,\n",
       "  0.0066991751082241535,\n",
       "  -6.462504825321957e-05,\n",
       "  -0.05613984540104866,\n",
       "  0.05322495847940445,\n",
       "  0.0022359222639352083,\n",
       "  -0.021985739469528198,\n",
       "  -0.02672739140689373,\n",
       "  0.020197425037622452,\n",
       "  0.07513511925935745,\n",
       "  -0.035484883934259415,\n",
       "  0.028822626918554306,\n",
       "  -0.027020258828997612,\n",
       "  0.04103274643421173,\n",
       "  0.09122151136398315,\n",
       "  0.026982422918081284,\n",
       "  -0.00011277346493443474,\n",
       "  0.047778766602277756,\n",
       "  0.00794275477528572,\n",
       "  0.034345466643571854,\n",
       "  0.01762368716299534,\n",
       "  -0.025831962004303932,\n",
       "  0.006831357255578041,\n",
       "  0.11059132218360901,\n",
       "  -0.015186501666903496,\n",
       "  0.049963150173425674,\n",
       "  -0.017848508432507515,\n",
       "  0.03086664341390133,\n",
       "  -0.007508426439017057,\n",
       "  0.045449867844581604,\n",
       "  -0.0587027370929718,\n",
       "  -0.040729109197854996,\n",
       "  -0.024549968540668488,\n",
       "  -0.019979970529675484,\n",
       "  0.050971731543540955,\n",
       "  -0.03182971104979515,\n",
       "  0.00750334607437253,\n",
       "  0.01859789900481701,\n",
       "  -0.025868378579616547,\n",
       "  -0.03700293228030205,\n",
       "  -0.012496686540544033,\n",
       "  -0.011270882561802864,\n",
       "  -0.004489682149142027,\n",
       "  -0.0885145291686058,\n",
       "  -0.1080537661910057,\n",
       "  0.004100131802260876,\n",
       "  -0.024510513991117477,\n",
       "  0.008399860002100468,\n",
       "  0.011766377836465836,\n",
       "  0.02180440165102482,\n",
       "  -0.014431959018111229,\n",
       "  0.03467494994401932,\n",
       "  -0.058218274265527725,\n",
       "  0.012324720621109009,\n",
       "  -0.023575013503432274,\n",
       "  0.011689843609929085,\n",
       "  -0.024557804688811302,\n",
       "  -0.0306258425116539,\n",
       "  0.02334428019821644,\n",
       "  -0.048935383558273315,\n",
       "  0.026916220784187317,\n",
       "  0.029866419732570648,\n",
       "  -0.08063715696334839,\n",
       "  0.041818901896476746,\n",
       "  -0.009288379922509193,\n",
       "  -0.0067566088400781155,\n",
       "  -0.01451076753437519,\n",
       "  0.017164673656225204,\n",
       "  -0.03530241176486015,\n",
       "  -0.03145453706383705,\n",
       "  -0.04587572440505028,\n",
       "  -0.05993197113275528,\n",
       "  0.0004112497845198959,\n",
       "  0.07444138824939728,\n",
       "  0.047800298780202866,\n",
       "  0.05894561484456062,\n",
       "  -0.09712149947881699,\n",
       "  -0.10323823988437653,\n",
       "  -0.004704223945736885,\n",
       "  0.030390577390789986,\n",
       "  0.005930012091994286,\n",
       "  -0.032592516392469406,\n",
       "  0.03726690635085106,\n",
       "  0.04921494051814079,\n",
       "  0.05208200961351395,\n",
       "  0.0010907913092523813,\n",
       "  0.043110597878694534,\n",
       "  0.02672078274190426,\n",
       "  0.01318393461406231,\n",
       "  -0.07506705075502396,\n",
       "  -0.03347068652510643,\n",
       "  -0.03141354024410248,\n",
       "  0.050599511712789536,\n",
       "  -0.007875503040850163,\n",
       "  0.01320816483348608,\n",
       "  -0.03410274162888527,\n",
       "  -0.047016508877277374,\n",
       "  -0.07218440622091293,\n",
       "  -0.005254123359918594,\n",
       "  -0.014212504960596561,\n",
       "  -0.014798527583479881,\n",
       "  -0.075481116771698,\n",
       "  0.001248626154847443,\n",
       "  0.0028126726392656565,\n",
       "  0.0070231202989816666,\n",
       "  -0.018816744908690453,\n",
       "  -0.03814244642853737,\n",
       "  0.038669403642416,\n",
       "  -0.05126436799764633,\n",
       "  -0.08393452316522598,\n",
       "  -0.04218350723385811,\n",
       "  -0.044320397078990936,\n",
       "  -0.007762732449918985,\n",
       "  -0.04561822488903999,\n",
       "  0.06090737506747246,\n",
       "  -0.06993749737739563,\n",
       "  0.05170052871108055,\n",
       "  -0.02664373070001602,\n",
       "  -0.06052147597074509,\n",
       "  -0.06488373130559921,\n",
       "  0.022549644112586975,\n",
       "  0.1170145571231842,\n",
       "  -0.020034445449709892,\n",
       "  0.039760030806064606,\n",
       "  -0.05028808116912842,\n",
       "  0.029706405475735664,\n",
       "  -0.0007561701931990683,\n",
       "  0.0668705478310585,\n",
       "  0.06005166098475456,\n",
       "  0.019600752741098404,\n",
       "  -0.026095915585756302,\n",
       "  -0.08237311989068985,\n",
       "  -0.07120492309331894,\n",
       "  -0.03230835497379303,\n",
       "  -0.00850074551999569,\n",
       "  0.04470153525471687,\n",
       "  -0.025179022923111916,\n",
       "  -0.02672233246266842,\n",
       "  -0.0005195355624891818,\n",
       "  -0.004355300217866898,\n",
       "  0.02207203581929207,\n",
       "  -0.06127742677927017,\n",
       "  0.003632528241723776,\n",
       "  0.04699184373021126,\n",
       "  0.04091031104326248,\n",
       "  0.026954902336001396,\n",
       "  0.024924075230956078,\n",
       "  0.10610134154558182,\n",
       "  0.06840702146291733,\n",
       "  -0.018298465758562088,\n",
       "  0.0679897591471672,\n",
       "  -0.06122373417019844,\n",
       "  0.0420626625418663,\n",
       "  -0.06468194723129272,\n",
       "  0.0388258621096611,\n",
       "  -0.035977475345134735,\n",
       "  0.00316184782423079,\n",
       "  -0.05502226576209068,\n",
       "  0.0036887649912387133,\n",
       "  0.033026035875082016,\n",
       "  -0.012183134444057941,\n",
       "  0.032471828162670135,\n",
       "  -0.002611053641885519,\n",
       "  -0.04321267828345299,\n",
       "  -0.082752525806427,\n",
       "  0.02243766374886036,\n",
       "  0.058102287352085114,\n",
       "  -0.02024371735751629,\n",
       "  -0.04735763743519783,\n",
       "  0.03387952223420143,\n",
       "  0.022740768268704414,\n",
       "  -0.017584266141057014,\n",
       "  0.03050813265144825,\n",
       "  -0.0038077097851783037,\n",
       "  -0.11578265577554703,\n",
       "  -0.013879401609301567,\n",
       "  0.03220343589782715,\n",
       "  0.1591539978981018,\n",
       "  0.048794519156217575,\n",
       "  -0.026415029540657997,\n",
       "  0.016187991946935654,\n",
       "  -0.054295387119054794,\n",
       "  0.03515414893627167,\n",
       "  0.12509077787399292,\n",
       "  0.010921798646450043,\n",
       "  -0.05765049159526825,\n",
       "  -0.029018566012382507,\n",
       "  0.03491559997200966,\n",
       "  -0.04960145056247711,\n",
       "  -0.031121689826250076,\n",
       "  -0.048641301691532135,\n",
       "  -0.0821344181895256,\n",
       "  0.006269358564168215,\n",
       "  -0.09349029511213303,\n",
       "  -0.030068034306168556,\n",
       "  0.00994799192994833,\n",
       "  0.0733526274561882,\n",
       "  -0.004065830260515213,\n",
       "  0.02225164882838726,\n",
       "  -0.01206603180617094,\n",
       "  -0.02917671389877796,\n",
       "  -0.003496437333524227,\n",
       "  0.040895916521549225,\n",
       "  -0.022811150178313255,\n",
       "  -0.014339711517095566,\n",
       "  0.09260017424821854,\n",
       "  -0.01251012459397316,\n",
       "  0.03202669322490692,\n",
       "  -0.0719839408993721,\n",
       "  0.0324031338095665,\n",
       "  -0.06738156825304031,\n",
       "  -0.0028889155946671963,\n",
       "  0.015469438396394253,\n",
       "  -0.04023635387420654,\n",
       "  0.026633426547050476,\n",
       "  0.012537418864667416,\n",
       "  0.07329786568880081,\n",
       "  0.042435795068740845,\n",
       "  0.031970031559467316,\n",
       "  0.020714564248919487,\n",
       "  -0.026811720803380013,\n",
       "  0.01961486041545868,\n",
       "  -0.04152116924524307,\n",
       "  -0.061849575489759445,\n",
       "  -0.06704414635896683,\n",
       "  -0.026246532797813416,\n",
       "  -0.06501781195402145,\n",
       "  -0.0871184840798378,\n",
       "  0.03152437135577202,\n",
       "  -0.01314247865229845,\n",
       "  -0.0973600298166275,\n",
       "  0.060924895107746124,\n",
       "  -0.016866013407707214,\n",
       "  0.05953431874513626,\n",
       "  0.0017192678060382605,\n",
       "  0.045271433889865875,\n",
       "  0.017204754054546356,\n",
       "  0.04099142178893089,\n",
       "  -0.003091138321906328,\n",
       "  0.06399713456630707,\n",
       "  -0.050633467733860016,\n",
       "  0.012899081222712994,\n",
       "  -0.06631828844547272,\n",
       "  0.006336959544569254,\n",
       "  -0.01689320057630539,\n",
       "  0.03915843740105629,\n",
       "  -0.004915426019579172,\n",
       "  0.029403552412986755,\n",
       "  -0.062134355306625366,\n",
       "  0.008213954046368599,\n",
       "  0.010039283894002438,\n",
       "  -0.019808286800980568,\n",
       "  -0.042555805295705795,\n",
       "  -0.011758428066968918,\n",
       "  -0.08583111315965652,\n",
       "  -0.06115078926086426,\n",
       "  0.024559777230024338,\n",
       "  -0.0328216515481472,\n",
       "  -0.03431062027812004,\n",
       "  0.03682801127433777,\n",
       "  -0.04175528511404991,\n",
       "  -0.06909109652042389,\n",
       "  0.05904894694685936,\n",
       "  -0.0009671786101534963,\n",
       "  -0.07380519807338715,\n",
       "  -0.05520394444465637,\n",
       "  -0.050658728927373886,\n",
       "  -0.08435318619012833,\n",
       "  -0.03665435314178467,\n",
       "  0.027599425986409187,\n",
       "  -0.0027247825637459755,\n",
       "  -0.05750029534101486,\n",
       "  0.07045142352581024,\n",
       "  0.004068974405527115,\n",
       "  0.02865174598991871,\n",
       "  0.05976295843720436,\n",
       "  -0.029506219550967216,\n",
       "  0.04876341670751572,\n",
       "  0.027430137619376183,\n",
       "  -0.014996052719652653,\n",
       "  0.026660753414034843,\n",
       "  -0.0534706749022007,\n",
       "  0.053507301956415176,\n",
       "  -0.044302042573690414,\n",
       "  0.06304668635129929,\n",
       "  -0.06364329904317856,\n",
       "  0.027741242200136185,\n",
       "  0.035155169665813446,\n",
       "  -0.029803622514009476,\n",
       "  -0.06677205115556717,\n",
       "  -0.08237474411725998,\n",
       "  0.0036693548317998648,\n",
       "  0.026248950511217117,\n",
       "  -0.07351749390363693,\n",
       "  -0.008242233656346798,\n",
       "  -0.025315620005130768,\n",
       "  -0.06499702483415604,\n",
       "  -0.022301528602838516,\n",
       "  0.034260377287864685,\n",
       "  0.0021985508501529694,\n",
       "  0.013125971890985966,\n",
       "  0.006112081464380026,\n",
       "  0.10069793462753296,\n",
       "  0.04225314408540726,\n",
       "  -0.01039137877523899,\n",
       "  -0.004997076001018286,\n",
       "  -0.10040666162967682,\n",
       "  -0.06894942373037338,\n",
       "  -9.646780381444842e-05,\n",
       "  0.018847569823265076,\n",
       "  0.0034488756209611893,\n",
       "  0.028925837948918343,\n",
       "  0.010435567237436771,\n",
       "  3.1105733796721324e-05,\n",
       "  -0.002606827300041914,\n",
       "  -0.019944289699196815,\n",
       "  -0.04343612119555473,\n",
       "  0.06031465902924538,\n",
       "  -0.007730160839855671,\n",
       "  0.034238919615745544,\n",
       "  0.09763974696397781,\n",
       "  0.010776224546134472,\n",
       "  -0.007350832689553499,\n",
       "  0.061622586101293564,\n",
       "  -0.016488023102283478,\n",
       "  0.030245967209339142,\n",
       "  -0.010284097865223885,\n",
       "  0.0025616681668907404,\n",
       "  -0.05481977388262749,\n",
       "  -0.04316744953393936,\n",
       "  0.00468814792111516,\n",
       "  -0.06859006732702255,\n",
       "  -0.02350418083369732,\n",
       "  -0.017822271212935448,\n",
       "  0.041956827044487,\n",
       "  0.022590572014451027,\n",
       "  -0.02863902598619461,\n",
       "  0.009872556664049625,\n",
       "  0.011802949011325836,\n",
       "  -0.1102726086974144,\n",
       "  -0.0019062119536101818,\n",
       "  0.06536117196083069,\n",
       "  0.07765129208564758,\n",
       "  0.03187059611082077,\n",
       "  0.03487187996506691,\n",
       "  0.0281063262373209,\n",
       "  0.05161067843437195,\n",
       "  -0.012188779190182686,\n",
       "  -0.03922712057828903,\n",
       "  0.01631467044353485,\n",
       "  -0.009087780490517616,\n",
       "  -0.0372125618159771,\n",
       "  0.04912416264414787,\n",
       "  0.02760489471256733,\n",
       "  0.011084935627877712,\n",
       "  -0.0036061524879187346,\n",
       "  0.01274216640740633,\n",
       "  0.0015076937852427363,\n",
       "  0.06313391029834747,\n",
       "  0.033683065325021744,\n",
       "  0.017946340143680573,\n",
       "  -0.01294436864554882,\n",
       "  0.06282127648591995,\n",
       "  -0.038650862872600555,\n",
       "  0.0671859085559845,\n",
       "  0.006586537230759859,\n",
       "  -1.3055957424512599e-05,\n",
       "  0.01757390983402729,\n",
       "  -0.0004038205661345273,\n",
       "  -0.07035853713750839,\n",
       "  -0.010478419251739979,\n",
       "  0.0004972164751961827,\n",
       "  -0.04879932478070259,\n",
       "  0.025469742715358734,\n",
       "  -0.0005429651937447488,\n",
       "  0.03466896340250969,\n",
       "  0.04381652921438217,\n",
       "  0.04273233562707901,\n",
       "  0.010173516348004341,\n",
       "  -0.028112933039665222,\n",
       "  0.021694447845220566,\n",
       "  0.001476959907449782,\n",
       "  -0.03473912924528122,\n",
       "  -0.0011069226311519742,\n",
       "  0.022281693294644356,\n",
       "  0.013537699356675148,\n",
       "  0.02903684787452221,\n",
       "  -0.05245966836810112,\n",
       "  -0.005741466302424669,\n",
       "  -0.05864690989255905,\n",
       "  -0.03595564514398575,\n",
       "  0.025359509512782097,\n",
       "  0.037775520235300064,\n",
       "  -0.04493860527873039,\n",
       "  -0.01778150163590908,\n",
       "  0.0061975582502782345,\n",
       "  -0.00262125744484365,\n",
       "  -0.02544468455016613,\n",
       "  0.022795027121901512,\n",
       "  -0.15417395532131195,\n",
       "  -0.018095500767230988,\n",
       "  -0.00024379367823712528,\n",
       "  0.039480507373809814,\n",
       "  0.007157085929065943,\n",
       "  -0.004248625133186579,\n",
       "  0.011769266799092293,\n",
       "  0.010865169577300549,\n",
       "  -0.016249751672148705,\n",
       "  1.1266960427747108e-05,\n",
       "  -0.03634585812687874,\n",
       "  0.028430994600057602,\n",
       "  -2.1108695364091545e-05,\n",
       "  0.02866288274526596,\n",
       "  -0.07438720017671585,\n",
       "  -0.03972702473402023,\n",
       "  0.04326412081718445,\n",
       "  0.02544897049665451,\n",
       "  0.025348270311951637,\n",
       "  0.029063405469059944,\n",
       "  0.04071425646543503,\n",
       "  -0.059724051505327225,\n",
       "  -0.011322631500661373,\n",
       "  0.06465347111225128,\n",
       "  0.0017285807989537716,\n",
       "  0.0544796921312809,\n",
       "  -0.06766196340322495,\n",
       "  -0.008276312611997128,\n",
       "  -0.03574991598725319,\n",
       "  -0.004175184294581413,\n",
       "  -0.02515602670609951,\n",
       "  0.01799033023416996,\n",
       "  0.032776035368442535,\n",
       "  -0.055802732706069946]}"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "inst_ds[-20]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'file_name': '../../input/validation_dataset_part1/video/000919.mp4',\n",
       " 'height': 960,\n",
       " 'width': 540,\n",
       " 'item_id': '000919',\n",
       " 'type': 'video',\n",
       " 'bbox': [99, 359, 318, 814],\n",
       " 'score': 0.9992340803146362,\n",
       " 'frame': 280,\n",
       " 'bbox_aspect': 0.48131868131868133,\n",
       " 'aspect_group': 3,\n",
       " 'feat': [0.10861119627952576,\n",
       "  -0.02023276872932911,\n",
       "  0.000677834206726402,\n",
       "  0.04453166946768761,\n",
       "  -0.08077826350927353,\n",
       "  0.013814250938594341,\n",
       "  -0.008523191325366497,\n",
       "  -0.0611051470041275,\n",
       "  -0.0397714264690876,\n",
       "  0.03194702789187431,\n",
       "  -0.0006806030287407339,\n",
       "  -0.019916066899895668,\n",
       "  0.07606640458106995,\n",
       "  0.009971742518246174,\n",
       "  -0.013884657062590122,\n",
       "  -0.021979905664920807,\n",
       "  0.08041248470544815,\n",
       "  -0.0171558428555727,\n",
       "  0.01628359965980053,\n",
       "  0.027200501412153244,\n",
       "  0.010065089911222458,\n",
       "  0.01591976173222065,\n",
       "  0.0682348683476448,\n",
       "  -0.04079398885369301,\n",
       "  0.0012808083556592464,\n",
       "  0.0553743839263916,\n",
       "  -0.01839214749634266,\n",
       "  0.08164402842521667,\n",
       "  -0.05385705828666687,\n",
       "  -0.019228944554924965,\n",
       "  0.02114301547408104,\n",
       "  -0.024600272998213768,\n",
       "  -0.054915424436330795,\n",
       "  -0.1035628691315651,\n",
       "  -0.007335315458476543,\n",
       "  0.02674494870007038,\n",
       "  -0.011918866075575352,\n",
       "  0.0051943291909992695,\n",
       "  -0.037380021065473557,\n",
       "  -0.044803548604249954,\n",
       "  -0.02558865025639534,\n",
       "  -0.057260558009147644,\n",
       "  0.0363578125834465,\n",
       "  -0.012578342109918594,\n",
       "  -0.05943750590085983,\n",
       "  -0.09108768403530121,\n",
       "  -0.014429399743676186,\n",
       "  -0.016254955902695656,\n",
       "  -0.04457131400704384,\n",
       "  -0.04249356687068939,\n",
       "  -0.02692616730928421,\n",
       "  0.09313350915908813,\n",
       "  0.01826678216457367,\n",
       "  -0.015197738073766232,\n",
       "  -0.07252022624015808,\n",
       "  -0.07828399538993835,\n",
       "  0.004812547471374273,\n",
       "  0.00643581710755825,\n",
       "  -0.05307281017303467,\n",
       "  -0.0702512264251709,\n",
       "  0.04165658354759216,\n",
       "  0.03468478098511696,\n",
       "  -0.028483951464295387,\n",
       "  -0.016901392489671707,\n",
       "  -0.002983693266287446,\n",
       "  -0.023870795965194702,\n",
       "  0.03681115433573723,\n",
       "  0.017926208674907684,\n",
       "  -0.0354781448841095,\n",
       "  -0.008896398358047009,\n",
       "  0.008050533011555672,\n",
       "  0.10579269379377365,\n",
       "  -0.015211955644190311,\n",
       "  -0.003988060634583235,\n",
       "  -0.023958537727594376,\n",
       "  0.0045199329033494,\n",
       "  -0.009660933166742325,\n",
       "  0.07028080523014069,\n",
       "  -0.008721746504306793,\n",
       "  0.003815128467977047,\n",
       "  0.0066991751082241535,\n",
       "  -6.462504825321957e-05,\n",
       "  -0.05613984540104866,\n",
       "  0.05322495847940445,\n",
       "  0.0022359222639352083,\n",
       "  -0.021985739469528198,\n",
       "  -0.02672739140689373,\n",
       "  0.020197425037622452,\n",
       "  0.07513511925935745,\n",
       "  -0.035484883934259415,\n",
       "  0.028822626918554306,\n",
       "  -0.027020258828997612,\n",
       "  0.04103274643421173,\n",
       "  0.09122151136398315,\n",
       "  0.026982422918081284,\n",
       "  -0.00011277346493443474,\n",
       "  0.047778766602277756,\n",
       "  0.00794275477528572,\n",
       "  0.034345466643571854,\n",
       "  0.01762368716299534,\n",
       "  -0.025831962004303932,\n",
       "  0.006831357255578041,\n",
       "  0.11059132218360901,\n",
       "  -0.015186501666903496,\n",
       "  0.049963150173425674,\n",
       "  -0.017848508432507515,\n",
       "  0.03086664341390133,\n",
       "  -0.007508426439017057,\n",
       "  0.045449867844581604,\n",
       "  -0.0587027370929718,\n",
       "  -0.040729109197854996,\n",
       "  -0.024549968540668488,\n",
       "  -0.019979970529675484,\n",
       "  0.050971731543540955,\n",
       "  -0.03182971104979515,\n",
       "  0.00750334607437253,\n",
       "  0.01859789900481701,\n",
       "  -0.025868378579616547,\n",
       "  -0.03700293228030205,\n",
       "  -0.012496686540544033,\n",
       "  -0.011270882561802864,\n",
       "  -0.004489682149142027,\n",
       "  -0.0885145291686058,\n",
       "  -0.1080537661910057,\n",
       "  0.004100131802260876,\n",
       "  -0.024510513991117477,\n",
       "  0.008399860002100468,\n",
       "  0.011766377836465836,\n",
       "  0.02180440165102482,\n",
       "  -0.014431959018111229,\n",
       "  0.03467494994401932,\n",
       "  -0.058218274265527725,\n",
       "  0.012324720621109009,\n",
       "  -0.023575013503432274,\n",
       "  0.011689843609929085,\n",
       "  -0.024557804688811302,\n",
       "  -0.0306258425116539,\n",
       "  0.02334428019821644,\n",
       "  -0.048935383558273315,\n",
       "  0.026916220784187317,\n",
       "  0.029866419732570648,\n",
       "  -0.08063715696334839,\n",
       "  0.041818901896476746,\n",
       "  -0.009288379922509193,\n",
       "  -0.0067566088400781155,\n",
       "  -0.01451076753437519,\n",
       "  0.017164673656225204,\n",
       "  -0.03530241176486015,\n",
       "  -0.03145453706383705,\n",
       "  -0.04587572440505028,\n",
       "  -0.05993197113275528,\n",
       "  0.0004112497845198959,\n",
       "  0.07444138824939728,\n",
       "  0.047800298780202866,\n",
       "  0.05894561484456062,\n",
       "  -0.09712149947881699,\n",
       "  -0.10323823988437653,\n",
       "  -0.004704223945736885,\n",
       "  0.030390577390789986,\n",
       "  0.005930012091994286,\n",
       "  -0.032592516392469406,\n",
       "  0.03726690635085106,\n",
       "  0.04921494051814079,\n",
       "  0.05208200961351395,\n",
       "  0.0010907913092523813,\n",
       "  0.043110597878694534,\n",
       "  0.02672078274190426,\n",
       "  0.01318393461406231,\n",
       "  -0.07506705075502396,\n",
       "  -0.03347068652510643,\n",
       "  -0.03141354024410248,\n",
       "  0.050599511712789536,\n",
       "  -0.007875503040850163,\n",
       "  0.01320816483348608,\n",
       "  -0.03410274162888527,\n",
       "  -0.047016508877277374,\n",
       "  -0.07218440622091293,\n",
       "  -0.005254123359918594,\n",
       "  -0.014212504960596561,\n",
       "  -0.014798527583479881,\n",
       "  -0.075481116771698,\n",
       "  0.001248626154847443,\n",
       "  0.0028126726392656565,\n",
       "  0.0070231202989816666,\n",
       "  -0.018816744908690453,\n",
       "  -0.03814244642853737,\n",
       "  0.038669403642416,\n",
       "  -0.05126436799764633,\n",
       "  -0.08393452316522598,\n",
       "  -0.04218350723385811,\n",
       "  -0.044320397078990936,\n",
       "  -0.007762732449918985,\n",
       "  -0.04561822488903999,\n",
       "  0.06090737506747246,\n",
       "  -0.06993749737739563,\n",
       "  0.05170052871108055,\n",
       "  -0.02664373070001602,\n",
       "  -0.06052147597074509,\n",
       "  -0.06488373130559921,\n",
       "  0.022549644112586975,\n",
       "  0.1170145571231842,\n",
       "  -0.020034445449709892,\n",
       "  0.039760030806064606,\n",
       "  -0.05028808116912842,\n",
       "  0.029706405475735664,\n",
       "  -0.0007561701931990683,\n",
       "  0.0668705478310585,\n",
       "  0.06005166098475456,\n",
       "  0.019600752741098404,\n",
       "  -0.026095915585756302,\n",
       "  -0.08237311989068985,\n",
       "  -0.07120492309331894,\n",
       "  -0.03230835497379303,\n",
       "  -0.00850074551999569,\n",
       "  0.04470153525471687,\n",
       "  -0.025179022923111916,\n",
       "  -0.02672233246266842,\n",
       "  -0.0005195355624891818,\n",
       "  -0.004355300217866898,\n",
       "  0.02207203581929207,\n",
       "  -0.06127742677927017,\n",
       "  0.003632528241723776,\n",
       "  0.04699184373021126,\n",
       "  0.04091031104326248,\n",
       "  0.026954902336001396,\n",
       "  0.024924075230956078,\n",
       "  0.10610134154558182,\n",
       "  0.06840702146291733,\n",
       "  -0.018298465758562088,\n",
       "  0.0679897591471672,\n",
       "  -0.06122373417019844,\n",
       "  0.0420626625418663,\n",
       "  -0.06468194723129272,\n",
       "  0.0388258621096611,\n",
       "  -0.035977475345134735,\n",
       "  0.00316184782423079,\n",
       "  -0.05502226576209068,\n",
       "  0.0036887649912387133,\n",
       "  0.033026035875082016,\n",
       "  -0.012183134444057941,\n",
       "  0.032471828162670135,\n",
       "  -0.002611053641885519,\n",
       "  -0.04321267828345299,\n",
       "  -0.082752525806427,\n",
       "  0.02243766374886036,\n",
       "  0.058102287352085114,\n",
       "  -0.02024371735751629,\n",
       "  -0.04735763743519783,\n",
       "  0.03387952223420143,\n",
       "  0.022740768268704414,\n",
       "  -0.017584266141057014,\n",
       "  0.03050813265144825,\n",
       "  -0.0038077097851783037,\n",
       "  -0.11578265577554703,\n",
       "  -0.013879401609301567,\n",
       "  0.03220343589782715,\n",
       "  0.1591539978981018,\n",
       "  0.048794519156217575,\n",
       "  -0.026415029540657997,\n",
       "  0.016187991946935654,\n",
       "  -0.054295387119054794,\n",
       "  0.03515414893627167,\n",
       "  0.12509077787399292,\n",
       "  0.010921798646450043,\n",
       "  -0.05765049159526825,\n",
       "  -0.029018566012382507,\n",
       "  0.03491559997200966,\n",
       "  -0.04960145056247711,\n",
       "  -0.031121689826250076,\n",
       "  -0.048641301691532135,\n",
       "  -0.0821344181895256,\n",
       "  0.006269358564168215,\n",
       "  -0.09349029511213303,\n",
       "  -0.030068034306168556,\n",
       "  0.00994799192994833,\n",
       "  0.0733526274561882,\n",
       "  -0.004065830260515213,\n",
       "  0.02225164882838726,\n",
       "  -0.01206603180617094,\n",
       "  -0.02917671389877796,\n",
       "  -0.003496437333524227,\n",
       "  0.040895916521549225,\n",
       "  -0.022811150178313255,\n",
       "  -0.014339711517095566,\n",
       "  0.09260017424821854,\n",
       "  -0.01251012459397316,\n",
       "  0.03202669322490692,\n",
       "  -0.0719839408993721,\n",
       "  0.0324031338095665,\n",
       "  -0.06738156825304031,\n",
       "  -0.0028889155946671963,\n",
       "  0.015469438396394253,\n",
       "  -0.04023635387420654,\n",
       "  0.026633426547050476,\n",
       "  0.012537418864667416,\n",
       "  0.07329786568880081,\n",
       "  0.042435795068740845,\n",
       "  0.031970031559467316,\n",
       "  0.020714564248919487,\n",
       "  -0.026811720803380013,\n",
       "  0.01961486041545868,\n",
       "  -0.04152116924524307,\n",
       "  -0.061849575489759445,\n",
       "  -0.06704414635896683,\n",
       "  -0.026246532797813416,\n",
       "  -0.06501781195402145,\n",
       "  -0.0871184840798378,\n",
       "  0.03152437135577202,\n",
       "  -0.01314247865229845,\n",
       "  -0.0973600298166275,\n",
       "  0.060924895107746124,\n",
       "  -0.016866013407707214,\n",
       "  0.05953431874513626,\n",
       "  0.0017192678060382605,\n",
       "  0.045271433889865875,\n",
       "  0.017204754054546356,\n",
       "  0.04099142178893089,\n",
       "  -0.003091138321906328,\n",
       "  0.06399713456630707,\n",
       "  -0.050633467733860016,\n",
       "  0.012899081222712994,\n",
       "  -0.06631828844547272,\n",
       "  0.006336959544569254,\n",
       "  -0.01689320057630539,\n",
       "  0.03915843740105629,\n",
       "  -0.004915426019579172,\n",
       "  0.029403552412986755,\n",
       "  -0.062134355306625366,\n",
       "  0.008213954046368599,\n",
       "  0.010039283894002438,\n",
       "  -0.019808286800980568,\n",
       "  -0.042555805295705795,\n",
       "  -0.011758428066968918,\n",
       "  -0.08583111315965652,\n",
       "  -0.06115078926086426,\n",
       "  0.024559777230024338,\n",
       "  -0.0328216515481472,\n",
       "  -0.03431062027812004,\n",
       "  0.03682801127433777,\n",
       "  -0.04175528511404991,\n",
       "  -0.06909109652042389,\n",
       "  0.05904894694685936,\n",
       "  -0.0009671786101534963,\n",
       "  -0.07380519807338715,\n",
       "  -0.05520394444465637,\n",
       "  -0.050658728927373886,\n",
       "  -0.08435318619012833,\n",
       "  -0.03665435314178467,\n",
       "  0.027599425986409187,\n",
       "  -0.0027247825637459755,\n",
       "  -0.05750029534101486,\n",
       "  0.07045142352581024,\n",
       "  0.004068974405527115,\n",
       "  0.02865174598991871,\n",
       "  0.05976295843720436,\n",
       "  -0.029506219550967216,\n",
       "  0.04876341670751572,\n",
       "  0.027430137619376183,\n",
       "  -0.014996052719652653,\n",
       "  0.026660753414034843,\n",
       "  -0.0534706749022007,\n",
       "  0.053507301956415176,\n",
       "  -0.044302042573690414,\n",
       "  0.06304668635129929,\n",
       "  -0.06364329904317856,\n",
       "  0.027741242200136185,\n",
       "  0.035155169665813446,\n",
       "  -0.029803622514009476,\n",
       "  -0.06677205115556717,\n",
       "  -0.08237474411725998,\n",
       "  0.0036693548317998648,\n",
       "  0.026248950511217117,\n",
       "  -0.07351749390363693,\n",
       "  -0.008242233656346798,\n",
       "  -0.025315620005130768,\n",
       "  -0.06499702483415604,\n",
       "  -0.022301528602838516,\n",
       "  0.034260377287864685,\n",
       "  0.0021985508501529694,\n",
       "  0.013125971890985966,\n",
       "  0.006112081464380026,\n",
       "  0.10069793462753296,\n",
       "  0.04225314408540726,\n",
       "  -0.01039137877523899,\n",
       "  -0.004997076001018286,\n",
       "  -0.10040666162967682,\n",
       "  -0.06894942373037338,\n",
       "  -9.646780381444842e-05,\n",
       "  0.018847569823265076,\n",
       "  0.0034488756209611893,\n",
       "  0.028925837948918343,\n",
       "  0.010435567237436771,\n",
       "  3.1105733796721324e-05,\n",
       "  -0.002606827300041914,\n",
       "  -0.019944289699196815,\n",
       "  -0.04343612119555473,\n",
       "  0.06031465902924538,\n",
       "  -0.007730160839855671,\n",
       "  0.034238919615745544,\n",
       "  0.09763974696397781,\n",
       "  0.010776224546134472,\n",
       "  -0.007350832689553499,\n",
       "  0.061622586101293564,\n",
       "  -0.016488023102283478,\n",
       "  0.030245967209339142,\n",
       "  -0.010284097865223885,\n",
       "  0.0025616681668907404,\n",
       "  -0.05481977388262749,\n",
       "  -0.04316744953393936,\n",
       "  0.00468814792111516,\n",
       "  -0.06859006732702255,\n",
       "  -0.02350418083369732,\n",
       "  -0.017822271212935448,\n",
       "  0.041956827044487,\n",
       "  0.022590572014451027,\n",
       "  -0.02863902598619461,\n",
       "  0.009872556664049625,\n",
       "  0.011802949011325836,\n",
       "  -0.1102726086974144,\n",
       "  -0.0019062119536101818,\n",
       "  0.06536117196083069,\n",
       "  0.07765129208564758,\n",
       "  0.03187059611082077,\n",
       "  0.03487187996506691,\n",
       "  0.0281063262373209,\n",
       "  0.05161067843437195,\n",
       "  -0.012188779190182686,\n",
       "  -0.03922712057828903,\n",
       "  0.01631467044353485,\n",
       "  -0.009087780490517616,\n",
       "  -0.0372125618159771,\n",
       "  0.04912416264414787,\n",
       "  0.02760489471256733,\n",
       "  0.011084935627877712,\n",
       "  -0.0036061524879187346,\n",
       "  0.01274216640740633,\n",
       "  0.0015076937852427363,\n",
       "  0.06313391029834747,\n",
       "  0.033683065325021744,\n",
       "  0.017946340143680573,\n",
       "  -0.01294436864554882,\n",
       "  0.06282127648591995,\n",
       "  -0.038650862872600555,\n",
       "  0.0671859085559845,\n",
       "  0.006586537230759859,\n",
       "  -1.3055957424512599e-05,\n",
       "  0.01757390983402729,\n",
       "  -0.0004038205661345273,\n",
       "  -0.07035853713750839,\n",
       "  -0.010478419251739979,\n",
       "  0.0004972164751961827,\n",
       "  -0.04879932478070259,\n",
       "  0.025469742715358734,\n",
       "  -0.0005429651937447488,\n",
       "  0.03466896340250969,\n",
       "  0.04381652921438217,\n",
       "  0.04273233562707901,\n",
       "  0.010173516348004341,\n",
       "  -0.028112933039665222,\n",
       "  0.021694447845220566,\n",
       "  0.001476959907449782,\n",
       "  -0.03473912924528122,\n",
       "  -0.0011069226311519742,\n",
       "  0.022281693294644356,\n",
       "  0.013537699356675148,\n",
       "  0.02903684787452221,\n",
       "  -0.05245966836810112,\n",
       "  -0.005741466302424669,\n",
       "  -0.05864690989255905,\n",
       "  -0.03595564514398575,\n",
       "  0.025359509512782097,\n",
       "  0.037775520235300064,\n",
       "  -0.04493860527873039,\n",
       "  -0.01778150163590908,\n",
       "  0.0061975582502782345,\n",
       "  -0.00262125744484365,\n",
       "  -0.02544468455016613,\n",
       "  0.022795027121901512,\n",
       "  -0.15417395532131195,\n",
       "  -0.018095500767230988,\n",
       "  -0.00024379367823712528,\n",
       "  0.039480507373809814,\n",
       "  0.007157085929065943,\n",
       "  -0.004248625133186579,\n",
       "  0.011769266799092293,\n",
       "  0.010865169577300549,\n",
       "  -0.016249751672148705,\n",
       "  1.1266960427747108e-05,\n",
       "  -0.03634585812687874,\n",
       "  0.028430994600057602,\n",
       "  -2.1108695364091545e-05,\n",
       "  0.02866288274526596,\n",
       "  -0.07438720017671585,\n",
       "  -0.03972702473402023,\n",
       "  0.04326412081718445,\n",
       "  0.02544897049665451,\n",
       "  0.025348270311951637,\n",
       "  0.029063405469059944,\n",
       "  0.04071425646543503,\n",
       "  -0.059724051505327225,\n",
       "  -0.011322631500661373,\n",
       "  0.06465347111225128,\n",
       "  0.0017285807989537716,\n",
       "  0.0544796921312809,\n",
       "  -0.06766196340322495,\n",
       "  -0.008276312611997128,\n",
       "  -0.03574991598725319,\n",
       "  -0.004175184294581413,\n",
       "  -0.02515602670609951,\n",
       "  0.01799033023416996,\n",
       "  0.032776035368442535,\n",
       "  -0.055802732706069946]}"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "inst_ds[-20]"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}
